import os

datapath ="/home/xun/dataset/Best_GEI_128/"
identity_list = sorted(os.listdir(datapath))
train_list = identity_list[0:50]
val_list = identity_list[50:74]
test_list = identity_list[74:]
resolution = 128
cut_padding = int(float(resolution)/64*10)
batch_size = (2,10)
frame_num=10
train_start_iteration = 0
hidden_dim = 256
margin = 0.2
lr = 0.0001
total_iter = 1200000
model_name = '6-CNN-avgcut-shared-independent-fc'
save_name = '6-CNN-avgcut-shared-independent-fc'
test_probe_condition_list = [['nm-05', 'nm-06'],['bg-01', 'bg-02'],['cl-01', 'cl-02']]
test_gallery_condition_list = ['nm-01', 'nm-02', 'nm-03', 'nm-04']
test_angle_list = ['000','018','036','054','072','090','108','126','144','162','180']
checkpoint_dir = './checkpoint/Latest_checkpoint_600000.pth.tar'
